package com.example.config.webscoket;


import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Auther: zhangr
 * @Date: 2019/09/11 11:48
 * @Description: websocket 服务类
 */

/**
 *
 * @ServerEndpoint 这个注解有什么作用？
 *
 * 这个注解用于标识作用在类上，它的主要功能是把当前类标识成一个WebSocket的服务端
 * 注解的值用户客户端连接访问的URL地址
 *
 */

@Slf4j
@Component
@ServerEndpoint("/websocket/{name}")
public class WebSocket {

    /**
     *  与某个客户端的连接对话，需要通过它来给客户端发送消息
     */
    private Session session;

    /**
     * 标识当前连接客户端的用户名
     */
    private String name;

    /**
     *  用于存所有的连接服务的客户端，这个对象存储是安全的
     */
    public static ConcurrentHashMap<String,WebSocket> webSocketSet = new ConcurrentHashMap<>();
    public static ConcurrentHashMap<String,WebSocket> webSocketEventSet = new ConcurrentHashMap<>();
    public static ConcurrentHashMap<String,WebSocket> webSocketWarningSet = new ConcurrentHashMap<>();


    @OnOpen
    public void OnOpen(Session session, @PathParam(value = "name") String name){
        this.session = session;
        this.name = name;
        // name是用来表示唯一客户端，如果需要指定发送，需要指定发送通过name来区分
        if(name.indexOf("event")>-1){
            webSocketEventSet.put(name,this);
        }else if(name.indexOf("warning")>-1){
            webSocketWarningSet.put(name,this);
        }else{
            webSocketSet.put(name,this);
        }
        log.info("[WebSocket] 连接成功，当前连接人数为：={}",webSocketSet.size());
    }


    @OnClose
    public void OnClose(){
        webSocketSet.remove(this.name);
        webSocketEventSet.remove(this.name);
        webSocketWarningSet.remove(this.name);
        log.info("[WebSocket] 退出成功，当前连接人数为：={}",webSocketSet.size());
    }

    @OnMessage
    public void OnMessage(String message){
        log.info("[WebSocket] 收到消息：{}",message);
        //判断是否需要指定发送，具体规则自定义
//        if(message.indexOf("TOUSER") == 0){
//            String name = message.substring(message.indexOf("TOUSER")+6,message.indexOf(";"));
//            AppointSending(name,message.substring(message.indexOf(";")+1,message.length()));
//        }else{
//            GroupSending(message);
//        }
//        if(message.equals("materials_heart")){
//            String mes = "success";
//            AppointSending(name,mes);
//        }
    }

    /**
     * 群发
     * @param message
     * @param type 推送类型
     */
    public void GroupSending(String message,String type){
        try {
            for (String name : webSocketSet.keySet()) {
                webSocketSet.get(name).session.getBasicRemote().sendText(message);
            }
            if (type.equals("event")) {
                for (String name : webSocketEventSet.keySet()) {
                    webSocketEventSet.get(name).session.getBasicRemote().sendText(message);
                }
            }else if(type.equals("warning")){
                for (String name : webSocketWarningSet.keySet()) {
                    webSocketWarningSet.get(name).session.getBasicRemote().sendText(message);
                }
            }
        }catch (Exception e){
            e.printStackTrace();
        }
    }

    /**
     * 指定发送
     * @param name
     * @param message
     */
    public void AppointSending(String name,String message){
        try {
            webSocketSet.get(name).session.getBasicRemote().sendText(message);
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}